package com.qbd.show.utils;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;


public class ZipUtils {

    /**
     *
     * 压缩文件夹或单个文件
     * @param srcFile 目录或者单个文件url
     * @param zipFile 压缩后的ZIP文件url
     * @throws Exception
     * @see [类、类#方法、类#成员]
     */
    public static void doCompress(String srcFile, String zipFile)
            throws Exception
    {
        doCompress(new File(srcFile), new File(zipFile));
    }

    /**
     *
     * 压缩文件夹或单个文件
     * @param srcFile 目录或者单个文件
     * @param destFile 压缩后的ZIP文件
     * @throws Exception
     * @see [类、类#方法、类#成员]
     */
    public static void doCompress(File srcFile, File destFile)
            throws Exception
    {
        ZipOutputStream out = new ZipOutputStream(new FileOutputStream(destFile));
        if (srcFile.isDirectory())
        {
            File[] files = srcFile.listFiles();
            for (File file : files)
            {
                doCompress(file, out);
            }
        }
        else
        {
            doCompress(srcFile, out);
        }
    }

    /**
     *
     * 压缩单个文件
     * @param pathname 文件完整url
     * @param out ZipOutputStream
     * @throws IOException
     * @see [类、类#方法、类#成员]
     */
    public static void doCompress(String pathname, ZipOutputStream out)
            throws IOException
    {
        doCompress(new File(pathname), out);
    }

    /**
     *
     * 压缩单个文件
     * @param file File实体类
     * @param out ZipOutputStream
     * @throws IOException
     * @see [类、类#方法、类#成员]
     */
    public static void doCompress(File file, ZipOutputStream out)
            throws IOException
    {
        if (file.exists())
        {
            byte[] buffer = new byte[1024];
            FileInputStream fis = new FileInputStream(file);
            out.putNextEntry(new ZipEntry(file.getName()));
            int len = 0;
            // 读取文件的内容,打包到zip文件
            while ((len = fis.read(buffer)) > 0)
            {
                out.write(buffer, 0, len);
            }
            out.flush();
            out.closeEntry();
            fis.close();
        }
    }
}
